'''
author:        Wang Chenyang <cy-wang21@mails.tsinghua.edu.cn>
date:          2025-07-30 17:02:50
Copyright © Department of Physics, Tsinghua University. All rights reserved
'''
import literature_format as lf
import PhysBERT
import numpy as np





def test_embedding():
    fname = "data/arXiv/T.Non-Hermitian_physics_and_open_systems.md"
    items = lf.markdown.parse_markdown_file(fname)
    print(len(items))

    bert = PhysBERT.BERTEmbedding()
    embed_items = get_embedding(bert, items)
    print(embed_items.shape)


if __name__ == "__main__":
    test_embedding()
